'use client';
import { cn } from '@lib/utils';
import { AnimatePresence, motion, type Transition, type Variants } from 'motion/react';
import { useMemo, useId } from 'react';

export type TextMorphProps = {
    children: string;
    as?: React.ElementType;
    className?: string;
    style?: React.CSSProperties;
    variants?: Variants;
    transition?: Transition;
};

export function TextMorph({
    children,
    as: Component = 'p',
    className,
    style,
    variants,
    transition,
}: TextMorphProps) {
    const uniqueId = useId();

    const characters = useMemo(() => {
        const charCounts: Record<string, number> = {};

        return children.split('').map((char) => {
            const lowerChar = char.toLowerCase();
            charCounts[lowerChar] = (charCounts[lowerChar] || 0) + 1;

            return {
                id: `${uniqueId}-${lowerChar}${charCounts[lowerChar]}`,
                label: char === ' ' ? '\u00A0' : char,
            };
        });
    }, [children, uniqueId]);

    const defaultVariants: Variants = {
        initial: { opacity: 0 },
        animate: { opacity: 1 },
        exit: { opacity: 0 },
    };

    const defaultTransition: Transition = {
        type: 'spring',
        stiffness: 280,
        damping: 18,
        mass: 0.3,
    };

    return (
        // @ts-expect-error - style is optional
        <Component className={cn(className)} aria-label={children} style={style}>
            <AnimatePresence mode='popLayout' initial={false}>
                {characters.map((character) => (
                    <motion.span
                        key={character.id}
                        layoutId={character.id}
                        className='inline-block'
                        aria-hidden='true'
                        initial='initial'
                        animate='animate'
                        exit='exit'
                        variants={variants || defaultVariants}
                        transition={transition || defaultTransition}
                    >
                        {character.label}
                    </motion.span>
                ))}
            </AnimatePresence>
        </Component>
    );
}
